"""
此为数据加载器，如果你需要使用你的数据，请自己根据数据格式进行修改
"""

from pathlib import Path

FLT_VWND = Path("/DatadiskExt/jjli/MRGTC/FLT_850_v")

class VwndLoader:
    def __init__(self, path: Path=FLT_VWND):
        self.path = path
        self.files = sorted(path.glob("*.nc"))
        self.index = 0

    def __iter__(self):
        return self

    def __next__(self):
        if self.index >= len(self.files):
            raise StopIteration
        file = self.files[self.index]
        self.index += 1
        return file
